package com.gtvtv.sparkdesksdk.model.response.agent;

import com.fasterxml.jackson.annotation.JsonProperty;

/**
 * Token使用统计
 */
public class TokenUsage {
    @JsonProperty("prompt_tokens")
    private int promptTokens; // 请求大模型的token
    
    @JsonProperty("completion_tokens")
    private int completionTokens; // 大模型回复的token
    
    @JsonProperty("total_tokens")
    private int totalTokens; // 总的token消耗

    public int getPromptTokens() {
        return promptTokens;
    }

    public void setPromptTokens(int promptTokens) {
        this.promptTokens = promptTokens;
    }

    public int getCompletionTokens() {
        return completionTokens;
    }

    public void setCompletionTokens(int completionTokens) {
        this.completionTokens = completionTokens;
    }

    public int getTotalTokens() {
        return totalTokens;
    }

    public void setTotalTokens(int totalTokens) {
        this.totalTokens = totalTokens;
    }
}